package cn.kent.middle;

/**
 * 2043. 简易银行系统
 * 银行共有 n 个账户，编号从 1 到 n 。
 * 每个账号的初始余额存储在一个下标从 0 开始的整数数组 balance 中，其中第 (i + 1) 个账户的初始余额是 balance[i] 。
 */
class Bank {
    public static void main(String[] args) {
        long[] balance = {10, 100, 20, 50, 30};
        Bank bank = new Bank(balance);
        System.out.println(bank.withdraw(3, 10));
        // 账户 3 余额为 $20 - $10 = $10 。
        System.out.println(bank.transfer(5, 1, 20));
        // 账户 5 的余额为 $30 - $20 = $10 ，账户 1 的余额为 $10 + $20 = $30 。
        System.out.println(bank.deposit(5, 20));
        // 账户 5 的余额为 $10 + $20 = $30 。
        System.out.println(bank.transfer(3, 4, 15));
        // 所以无法转账 $15 。
        System.out.println(bank.withdraw(10, 50));
    }

    /**
     * 92ms 100%
     */

    private long[] balance;

    //使用下标从 0 开始的整数数组 balance 初始化该对象。
    public Bank(long[] balance) {
        this.balance = balance;
    }

    //从编号为 account1 的账户向编号为 account2 的账户转帐 money 美元。如果交易成功，返回 true ，否则，返回 false。
    public boolean transfer(int account1, int account2, long money) {
        // 账号1和2是否存在，账号1余额是否够扣
        if (account1 > balance.length || account2 > balance.length || balance[account1 - 1] < money) {
            return false;
        }
        balance[account1 - 1] -= money;
        balance[account2 - 1] += money;
        return true;
    }

    //向编号为 account 的账户存款 money 美元。如果交易成功，返回 true ；否则，返回 false 。
    public boolean deposit(int account, long money) {
        if (account > this.balance.length) {
            return false;
        }
        balance[account - 1] += money;
        return true;
    }

    //从编号为 account 的账户取款 money 美元。如果交易成功，返回 true ；否则，返回 false 。
    public boolean withdraw(int account, long money) {
        if (account > balance.length || balance[account - 1] < money) {
            return false;
        }
        balance[account - 1] -= money;
        return true;
    }
}